import torch
from utils.preprocessing import do_SWA


resumes = [
    '/media/gserver/extra/FashionAI/round2/NAS[1234][r2]/weights-13-2888-[0.8672]-[0.9625].pth',
    '/media/gserver/extra/FashionAI/round2/NAS[1234][r2]/weights-12-1666-[0.8672]-[0.9619].pth',
    '/media/gserver/extra/FashionAI/round2/NAS[1234][r2]/weights-10-2222-[0.8672]-[0.9618].pth',
]
SWA_weights = do_SWA(resumes=resumes)
torch.save(SWA_weights, '/media/gserver/extra/FashionAI/round2/NAS[1234][r2]/SWA[12-13].pth')
